import folium
import pandas as pd

beijing_map = folium.Map(location=[39.9042, 116.4074], zoom_start=12)

# 导入excel的坐标数据，只读取前200行
df = pd.read_excel("E:\\大二上\\微观经济学\\data\\路侧停车位基础信息.xlsx", nrows=200)

# 加载北京边界的GeoJSON文件
geojson_path = "E:\\大二上\\Weiguan\\110000.geojson"
folium.GeoJson(geojson_path, name="Beijing Boundaries").add_to(beijing_map)

# 表格中每一列依次代表 停车场名、位置、经度、纬度
# 将数据框转换为字典格式
parking_lots = df.to_dict(orient='records')

#print(data_dict)

# 将停车场信息添加到地图上
for lot in parking_lots:
    folium.Marker(
        location=[lot["纬度"], lot["经度"]],
        popup=lot["停车场名"] + "泊位数量：" + str(lot["泊位数量"]),
        icon=folium.Icon(color='blue', icon='info-sign')
    ).add_to(beijing_map)

beijing_map.save("beijing_parking_lots_map.html")
